import csv
import sqlite3

connection = sqlite3.connect('../db.sqlite3')
cursor = connection.cursor()

create_table = '''
create table data_Display
(
    Name text not null,
    ID text not null,
    PID text not null,
    Platform text not null,
    Price text,
    Shape text,
    Size text,
    Resolution text,
    RefreshRate text,
    Panel text,
    ColorGamut text,
    Curvature text,
    constraint display_pk
        primary key (Name,ID,PID,Platform)
);


'''

cursor.execute(create_table)
file = open('../GetInfo/Details/DisplayInfo.csv', encoding='utf-8')
contents = csv.reader(file)
insert_records = "INSERT INTO data_Display (Name,ID,PID,Platform,Price,Shape,Size,Resolution,RefreshRate,Panel,ColorGamut,Curvature) " \
                 "VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"

cursor.executemany(insert_records, contents)
select_all = "SELECT * FROM data_Display"
rows = cursor.execute(select_all).fetchall()
for r in rows:
    print(r)

# Committing the changes
connection.commit()
connection.close()
